from os import path
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import jieba

from wordcloud import WordCloud

# Windows下面文件路径要使用双斜线，第一个斜线表示转义


# 读取语料数据文件
# 含有中文的文件要使用“rb”以二进制格式读取，否则会出现编码错误
def jb(text):

    import os
    # print('****************')
    path=os.getcwd()

    # text = open('222.txt', "rb").read()

    # 使用jieba进行中文分词
    mytext = " ".join(jieba.cut(text))


    # 读取蒙版图片
    alice_mask = np.array(Image.open(path+"/static/img/touxiang.png"))

    # 暂定最大词汇数为1000000
    wc = WordCloud(mask=alice_mask, font_path=path+'/static/simhei.ttf',
                   contour_width=3, contour_color="steelblue")

    # 生成词云
    wc.generate(mytext)

    # 存储到文件
    wc.to_file(path+"/static/img/meng.png")

    # 展示词云结果
    # plt.imshow(wc, interpolation='bilinear')
    # plt.axis("off")
    # plt.figure()
    # plt.imshow(alice_mask, cmap=plt.cm.gray, interpolation='bilinear')
    # # plt.axis("off")
    # # plt.show()
    return 'static/img/meng.png'